package com.gxc.string;

import java.util.*;
import java.util.stream.Collectors;

/**
 * 49. 字母异位词分组
 给你一个字符串数组，请你将 字母异位词 组合在一起。可以按任意顺序返回结果列表。

 字母异位词 是由重新排列源单词的所有字母得到的一个新单词。
 */
public class GroupAnagrams {

    public static void main(String[] args) {

    }

    /**
     * 通过给 str 排序，获得相同的key
     * @param strs
     * @return
     */
    public List<List<String>> groupAnagrams(String[] strs) {
        List<List<String>> res = new ArrayList<>();
        Map<String, List<String>> map = new HashMap<>();
        for (int i = 0; i < strs.length; i++) {
            String str = strs[i];
            char[] chars = str.toCharArray();
            Arrays.sort(chars);

            String key = new String(chars);
            if (map.keySet().contains(key)) {
                map.getOrDefault(key, new ArrayList<>()).add(str);
            } else {
                List<String> list = new ArrayList<>();
                list.add(str);
                map.put(key, list);
            }
        }
        return map.values().stream().collect(Collectors.toList());
    }

    /**
     * 通过 把 str 转换成 数组计数，转换成相同的key
     * @param strs
     * @return
     */
    public List<List<String>> groupAnagrams2(String[] strs) {
        List<List<String>> res = new ArrayList<>();
        Map<String, List<String>> map = new HashMap<>();
        for (int i = 0; i < strs.length; i++) {
            String str = strs[i];
            char[] chars = str.toCharArray();
            int[] arr = new int[26];
            for (int j = 0; j < chars.length; j++) {
                arr[chars[j] - 'a'] = arr[chars[j] - 'a']+1;
            }

            StringBuilder sb = new StringBuilder();
            for (int j = 0; j < arr.length; j++) {
                if (arr[j] != 0) {
                    //sb.append(j).append(arr[j]);  错误  21 2 和  2 12
                    sb.append((char)(j + 'a')).append(arr[j]);
                }
            }

            String key = sb.toString();
            List<String> list = map.getOrDefault(key, new ArrayList<>());
            list.add(str);
            map.put(key, list);
        }
        return new ArrayList<>(map.values());
    }
}
